root=$(
  cd "$(dirname $0)/.."
  pwd
)

############################# 閸忔娊鏁柊宥囩枂 ############################
config="c5B"
other=""
cmd="python laboratory.py "$config" "$other
device=2

name="wxy_train_sweet_add_"$config"_GPU"$device
##################################################################
docker rm -f $name

version=$(docker --version)
v19="*version 19*"
if test -z $(which nvidia-smi); then
  gpu=""
else
  if [[ $version == $v19 ]]; then
    gpu="--gpus all"
  else
    gpu="--runtime nvidia"
  fi
fi

echo "==================================================================="
echo "  警告!  使用2次快捷键 ctrl+p, ctrl+q 退出"
echo "  警告!  不能用 ctrl+c !"
echo "==================================================================="

docker run -it $gpu \
  --log-opt max-size=10m --log-opt max-file=5 \
  -v $root/:$root \
  -v /data/wxy:/data/wxy \
  -w $root \
  -e PYTHONUNBUFFERED=1 \
  --net=host \
  --name $name \
  -e CUDA_VISIBLE_DEVICES=$device \
  192.168.31.103:5000/wangxiaoyu/pytorch1.2:v7 \
  $cmd


# 这个镜像无法用GPU
# 192.168.31.103:5000/wangxiaoyu/pytorch1.6:v3
